package org.mxp.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.ReturnedMessage;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

@Slf4j
@Component
public class MyCallBack implements RabbitTemplate.ConfirmCallback, RabbitTemplate.ReturnsCallback {

    // 注入RabbitTemplate，用于全局设置他的某些回调
    @Autowired
    private RabbitTemplate rabbitTemplate;

    // Bean初始化完成后被执行初始化
    @PostConstruct
    public void init() {
        // 设置发布确认回调（找不到交换机时的回调）
        rabbitTemplate.setConfirmCallback(this);
        // 设置回退消息回调（找不到队列时的回调）
        rabbitTemplate.setReturnsCallback(this);
    }

    /**
     * 交换机不管是否收到消息都会回调该方法
     * @param correlationData  保存回调信息的Id及相关信息，需要在发送消息时由生产者设置
     * @param ack              交换机是否收到消息 true/false
     * @param cause            未收到消息的原因 只有ack为false才有
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        String id = correlationData != null ? correlationData.getId() : "";
        if (ack) {
            log.info("{} 交换机已收到消息", id);
        } else {
            log.info("{} 交换机未收到消息，未收到消息原因为：{}", id, cause);
        }
    }

    /**
     * 当消息无法路由的时候的回调方法（队列不存在，或者路由键不存在之类的，反正到不了队列就会触发）
     *  message      消息
     *  replyCode    编码
     *  replyText    退回原因
     *  exchange     从哪个交换机退回
     *  routingKey   通过哪个路由 key 退回
     */
    @Override
    public void returnedMessage(ReturnedMessage returned) {
        log.info("消息：{}，被交换机{}退回，原因：{}，路由key：{}，code:{}",
                new String(returned.getMessage().getBody()), returned.getExchange(),
                returned.getReplyText(), returned.getRoutingKey(),
                returned.getReplyCode());
    }
}
